package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

/**
 * 聊天室服务端
 */
public class Server {
    /*
        java.net.ServerSocket
        运行在服务端的ServerSocket主要有两个工作:
        1:向系统申请服务端口，客户端就可以通过这个端口与服务端建立链接了
        2:监听服务端口，一旦客户端建立链接就会创建一个Socket用于与该客户端交互
        可以将ServerSocket比喻为某些客服中心的"服务总机"
     */
    private ServerSocket serverSocket;
    /*
        存放所有客户端的输出流,用于广播
     */
    private List<PrintWriter> allOut = new ArrayList<>();

    public Server(){
        try {
            System.out.println("正在启动服务端...");
            serverSocket = new ServerSocket(8088);
            System.out.println("服务端启动完毕!");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    public void start(){
        try {
            while(true) {
                System.out.println("等待客户端链接...");
                /*
                ServerSocket定义的重要方法:
                Socket accept()
                用于监听服务端口，等待客户端的链接，一旦一个客户端链接上则会立即返回
                一个Socket实例，通过这个Socket就可以与链接上的客户端交互了
                该方法是一个阻塞方法，调用后会"卡住"，直到一个客户端建立链接。
             */
                Socket socket = serverSocket.accept();
                System.out.println("一个客户端链接了!");
                //启动一个线程来处于与该客户端的交互
                ClientHandler handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();

            }
        } catch (IOException e) {
            e.printStackTrace();
        }

    }
    public static void main(String[] args) {
        Server server = new Server();
        server.start();
    }

    /**
     * 该线程任务负责与指定客户端进行交互
     */
    private class ClientHandler implements Runnable{
        private Socket socket;
        private String host;//记录客户端的IP地址

        public ClientHandler(Socket socket){
            this.socket = socket;
            //通过socket获取远端计算机的地址信息
            host = socket.getInetAddress().getHostAddress();
        }

        public void run(){
            PrintWriter pw = null;
            try {
                InputStream in = socket.getInputStream();
                InputStreamReader isr = new InputStreamReader(in);
                BufferedReader br = new BufferedReader(isr);

                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out, StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw,true);

                /*
                    确保多个线程不能同时向allOut集合中添加元素,避免并发安全问题
                    通常:可以选取临界资源作为同步监视器对象
                 */
                synchronized (allOut) {
                    //将该输出流存入共享集合allOut中,便于广播消息
                    allOut.add(pw);
                }

                //广播该客户端上线了
                sendMessage(host + "上线了,当前在线人数:" + allOut.size());


                String line;
                while ((line = br.readLine()) != null) {
                    sendMessage(host + "说:" + line);
                }
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                //处理客户端断开后的操作
                synchronized (allOut) {
                    allOut.remove(pw);//将该客户端输出流从共享集合中删除
                }
                sendMessage(host+"下线了,当前在线人数:"+allOut.size());
            }
        }

        /**
         * 将消息广播给所有客户端
         * @param line
         */
        private void sendMessage(String line){
            System.out.println(line);
            synchronized (allOut) {
                for (PrintWriter o : allOut) {
                    o.println(line);
                }
            }
        }

    }

}
